Python3读写CSV文件

2024-06-03 16:14| 来源: 网络整理| 查看: 265

CSV(Comma-Separated Values)即逗号分隔值，一种以逗号分隔按行存储的文本文件，所有的值都表现为字符串类型（注意：数字为字符串类型）。

如果CSV中有中文，应以utf-8编码读写，如果要支持Excel查看，应是要用utf-8 with bom格式及utf-8-sig

Python3操作CSV文件使用自带的csv包 reader=csv.reader(f, delimiter=',')：用来读取数据，reader为生成器，每次读取一行，每行数据为列表格式，可以通过delimiter参数指定分隔符 writer=csv.writer(f)：用来写入数据，按行写入，writer支持writerow(列表)单行写入，和writerows(嵌套列表)批量写入多行，无须手动保存。

当文件中有标题行时，可以使用header=next(reader)先获取到第一行的数据，再进行遍历所有的数据行。写入时，可以先使用writer.writerow(标题行列表)，写入标题行，再使用writer.writerows(多行数据嵌套列表)，写入多行数据（也可以逐行写入）。

数据文件data.csv:

name,password abc,123456 张五,123#456 张#abc123,123456 666,123456 a b,123456

读取示例：

import csv with open('data.csv', encoding='utf-8') as f: reader = csv.reader(f) header = next(reader) print(header) for row in reader: print(row)

注意：reader必须在文件打开的上下文中使用，否则文件被关闭后reader无法使用所有的数字被作为字符串，如果要使用数字格式，应使用int()/float()做相应转换

写入示例：

import csv header = ['name', 'password', 'status'] data = [ ['abc', '123456', 'PASS'], ['张五', '123#456', 'PASS'], ['张#abc123', '123456', 'PASS'], ['666', '123456', 'PASS'], ['a b', '123456', 'PASS'] ] with open('result.csv', 'w', encoding='utf-8', newline='') as f: writer = csv.writer(f) writer.writerow(header) writer.writerows(data)

注意，打开文件时应指定格式为w, 文本写入，不支持wb,二进制写入，当然，也可以使用a/w+/r+ 打开文件时，指定不自动添加新行newline='',否则每写入一行就或多一个空行。如果想写入的文件Excel打开没有乱码，utf-8可以改为utf-8-sig。

结果展示，result.csv内容：：

name,password,status abc,123456,PASS 张五,123#456,PASS 张#abc123,123456,PASS 666,123456,PASS a b,123456,PASS 使用字典格式的数据：DictReader, DictWriter

注意数据必须有标题行时才能使用

reader=csv.DictReader(f)：直接将标题和每一列数据组装成有序字典（OrderedDict）格式，无须再单独读取标题行 writer=csv.DictWriter(f, 标题行列表)：写入时可使用writer.writeheader()写入标题，然后使用writer.writerow(字典格式数据行)或write.writerows(多行数据)

读取示例：

import csv with open('data.csv', encoding='utf-8') as f: reader = csv.DictReader(f) for row in reader: print(row['name'], row['password'])

写入示例：

import csv header = ['name', 'password', 'status'] data = [ {'name':'abc', 'password':'123456', 'status':'PASS'}, {'name':'张五', 'password':'123#456', 'status':'PASS'}, {'name':'张#abc123', 'password':'123456', 'status':'PASS'}, {'name':'666', 'password':'123456', 'status':'PASS'}, {'name':'a b', 'password':'123456', 'status':'PASS'} ] with open('result2.csv', 'w', encoding='utf-8', newline='') as f: writer = csv.DictWriter(f, header) writer.writeheader() writer.writerows(data)

写入结果，同result.csv

参考链接：https://python3-cookbook.readthedocs.io/zh_CN/latest/c06/p01_read_write_csv_data.html

【本文地址】

公司简介

联系我们

今日新闻

推荐新闻

专题文章